// Copyright 2014 PDFium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

// Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com

#include "../../public/fpdf_ext.h"
#include "../../third_party/base/nonstd_unique_ptr.h"
#include "../include/fsdk_define.h"
#include "../include/fsdk_mgr.h"
#include "../include/formfiller/FFL_FormFiller.h"
#include "../include/javascript/IJavaScript.h"

#if _FX_OS_ == _FX_ANDROID_
#include "time.h"
#else
#include <ctime>
#endif

//extern CPDFDoc_Environment* g_pFormFillApp;
class CFX_SystemHandler:public IFX_SystemHandler
{
public:
    CFX_SystemHandler(CPDFDoc_Environment* pEnv):m_pEnv(pEnv),m_nCharSet(-1) {}
public:
    virtual void                InvalidateRect(FX_HWND hWnd, FX_RECT rect) ;
    virtual void                OutputSelectedRect(void* pFormFiller, CPDF_Rect& rect);

    virtual FX_BOOL             IsSelectionImplemented();

    virtual CFX_WideString      GetClipboardText(FX_HWND hWnd){return L"";}
    virtual FX_BOOL             SetClipboardText(FX_HWND hWnd, CFX_WideString string) {return FALSE;}

    virtual void                ClientToScreen(FX_HWND hWnd, int32_t& x, int32_t& y) {}
    virtual void                ScreenToClient(FX_HWND hWnd, int32_t& x, int32_t& y) {}

    /*cursor style
    FXCT_ARROW
    FXCT_NESW
    FXCT_NWSE
    FXCT_VBEAM
    FXCT_HBEAM
    FXCT_HAND
    */
    virtual void                SetCursor(int32_t nCursorType);

    virtual FX_HMENU            CreatePopupMenu() {return NULL;}
    virtual FX_BOOL             AppendMenuItem(FX_HMENU hMenu, int32_t nIDNewItem, CFX_WideString string) {return FALSE;}
    virtual FX_BOOL             EnableMenuItem(FX_HMENU hMenu, int32_t nIDItem, FX_BOOL bEnabled) {return FALSE;}
    virtual int32_t         TrackPopupMenu(FX_HMENU hMenu, int32_t x, int32_t y, FX_HWND hParent) {return -1;}
    virtual void                DestroyMenu(FX_HMENU hMenu) {}

    virtual CFX_ByteString      GetNativeTrueTypeFont(int32_t nCharset);
    virtual FX_BOOL             FindNativeTrueTypeFont(int32_t nCharset, CFX_ByteString sFontFaceName);
    virtual CPDF_Font*          AddNativeTrueTypeFontToPDF(CPDF_Document* pDoc, CFX_ByteString sFontFaceName, uint8_t nCharset);

    virtual int32_t         SetTimer(int32_t uElapse, TimerCallback lpTimerFunc) ;
    virtual void                KillTimer(int32_t nID) ;


    virtual FX_BOOL             IsSHIFTKeyDown(FX_DWORD nFlag) {return m_pEnv->FFI_IsSHIFTKeyDown(nFlag);}
    virtual FX_BOOL             IsCTRLKeyDown(FX_DWORD nFlag) {return m_pEnv->FFI_IsCTRLKeyDown(nFlag);}
    virtual FX_BOOL             IsALTKeyDown(FX_DWORD nFlag) {return m_pEnv->FFI_IsALTKeyDown(nFlag);}
    virtual FX_BOOL             IsINSERTKeyDown(FX_DWORD nFlag) {return m_pEnv->FFI_IsINSERTKeyDown(nFlag);}

    virtual FX_SYSTEMTIME       GetLocalTime();

    virtual int32_t         GetCharSet() {return m_nCharSet;}
    virtual void                SetCharSet(int32_t nCharSet) {m_nCharSet = nCharSet;}
private:
    CPDFDoc_Environment* m_pEnv;
    int     m_nCharSet;
};

void CFX_SystemHandler::SetCursor(int32_t nCursorType)
{

    m_pEnv->FFI_SetCursor(nCursorType);
}

void CFX_SystemHandler::InvalidateRect(FX_HWND hWnd, FX_RECT rect)
{
    //g_pFormFillApp->FFI_Invalidate();
    CPDFSDK_Annot* pSDKAnnot = (CPDFSDK_Annot*)hWnd;
    CPDF_Page* pPage = NULL;
    CPDFSDK_PageView* pPageView = NULL;
    pPageView = pSDKAnnot->GetPageView();
    pPage = pSDKAnnot->GetPDFPage();
    if(!pPage || !pPageView)
        return;
    CPDF_Matrix page2device;
    pPageView->GetCurrentMatrix(page2device);
    CPDF_Matrix device2page;
    device2page.SetReverse(page2device);
    FX_FLOAT left, top, right,bottom;
    device2page.Transform((FX_FLOAT)rect.left, (FX_FLOAT)rect.top, left, top);
    device2page.Transform((FX_FLOAT)rect.right, (FX_FLOAT)rect.bottom, right, bottom);
//  m_pEnv->FFI_DeviceToPage(pPage, rect.left, rect.top, (double*)&left, (double*)&top);
//  m_pEnv->FFI_DeviceToPage(pPage, rect.right, rect.bottom, (double*)&right, (double*)&bottom);
    CPDF_Rect rcPDF(left, bottom, right, top);
    rcPDF.Normalize();

    m_pEnv->FFI_Invalidate(pPage, rcPDF.left, rcPDF.top, rcPDF.right, rcPDF.bottom);
}
void CFX_SystemHandler::OutputSelectedRect(void* pFormFiller, CPDF_Rect& rect)
{
    CFFL_FormFiller* pFFL = (CFFL_FormFiller*)pFormFiller;
    if(pFFL)
    {
        CPDF_Point leftbottom = CPDF_Point(rect.left, rect.bottom);
        CPDF_Point righttop = CPDF_Point(rect.right, rect.top);
        CPDF_Point ptA = pFFL->PWLtoFFL(leftbottom);
        CPDF_Point ptB = pFFL->PWLtoFFL(righttop);


        CPDFSDK_Annot* pAnnot  = pFFL->GetSDKAnnot();
        ASSERT(pAnnot);
        CPDF_Page* pPage = pAnnot->GetPDFPage();
        ASSERT(pPage);
        m_pEnv->FFI_OutputSelectedRect(pPage, ptA.x, ptB.y, ptB.x, ptA.y);
    }

}

FX_BOOL CFX_SystemHandler::IsSelectionImplemented()
{
    if(m_pEnv)
    {
        FPDF_FORMFILLINFO* pInfo = m_pEnv->GetFormFillInfo();
        if(pInfo && pInfo->FFI_OutputSelectedRect)
            return TRUE;
    }
    return FALSE;
}

CFX_ByteString CFX_SystemHandler::GetNativeTrueTypeFont(int32_t nCharset)
{
    return "";
}

FX_BOOL CFX_SystemHandler::FindNativeTrueTypeFont(int32_t nCharset, CFX_ByteString sFontFaceName)
{
    CFX_FontMgr* pFontMgr = CFX_GEModule::Get()->GetFontMgr();
    if(pFontMgr)
    {
        CFX_FontMapper* pFontMapper = pFontMgr->m_pBuiltinMapper;
        if(pFontMapper)
        {
            int nSize = pFontMapper->m_InstalledTTFonts.GetSize();
            if(nSize ==0)
            {
                pFontMapper->LoadInstalledFonts();
                nSize = pFontMapper->m_InstalledTTFonts.GetSize();
            }

            for(int i=0; i<nSize; i++)
            {
                if(pFontMapper->m_InstalledTTFonts[i].Compare(sFontFaceName))
                    return TRUE;
            }
        }

    }

    return FALSE;
}

static int CharSet2CP(int charset)
{
    if (charset == 128)
        return 932;
    if (charset == 134)
        return 936;
    if (charset == 129)
        return 949;
    if (charset == 136)
        return 950;
    return 0;
}
CPDF_Font* CFX_SystemHandler::AddNativeTrueTypeFontToPDF(CPDF_Document* pDoc, CFX_ByteString sFontFaceName,
                                                         uint8_t nCharset)
{
    if(pDoc)
    {
        CFX_Font* pFXFont = new CFX_Font();
        pFXFont->LoadSubst(sFontFaceName,TRUE,0,0,0,CharSet2CP(nCharset),FALSE);
        CPDF_Font* pFont = pDoc->AddFont(pFXFont,nCharset,FALSE);
        delete pFXFont;
        return pFont;
    }

    return NULL;
}


int32_t CFX_SystemHandler::SetTimer(int32_t uElapse, TimerCallback lpTimerFunc)
{
    return m_pEnv->FFI_SetTimer(uElapse, lpTimerFunc);
}
void CFX_SystemHandler::KillTimer(int32_t nID)
{
    m_pEnv->FFI_KillTimer(nID);
}

FX_SYSTEMTIME CFX_SystemHandler::GetLocalTime()
{
    return m_pEnv->FFI_GetLocalTime();
}


CJS_RuntimeFactory* GetJSRuntimeFactory()
{
#ifdef _V8_SUPPORT_
    static CJS_RuntimeFactory s_JSRuntimeFactory;
    return &s_JSRuntimeFactory;
#else
    return NULL;
#endif
}

CPDFDoc_Environment::CPDFDoc_Environment(CPDF_Document* pDoc, FPDF_FORMFILLINFO* pFFinfo) :
    m_pAnnotHandlerMgr(NULL),
    m_pActionHandler(NULL),
    m_pJSRuntime(NULL),
    m_pInfo(pFFinfo),
    m_pSDKDoc(NULL),
    m_pPDFDoc(pDoc),
    m_pIFormFiller(NULL)
{
    m_pSysHandler = new CFX_SystemHandler(this);

    m_pJSRuntimeFactory = NULL;
#ifdef _V8_SUPPORT_
    m_pJSRuntimeFactory = GetJSRuntimeFactory();
    m_pJSRuntimeFactory->AddRef();
#endif
}

CPDFDoc_Environment::~CPDFDoc_Environment()
{
    delete m_pIFormFiller;
    m_pIFormFiller = NULL;
#ifdef _V8_SUPPORT_
    if (m_pJSRuntime && m_pJSRuntimeFactory)
        m_pJSRuntimeFactory->DeleteJSRuntime(m_pJSRuntime);
    m_pJSRuntimeFactory->Release();
#endif

    delete m_pSysHandler;
    m_pSysHandler = NULL;

    delete m_pAnnotHandlerMgr;
    m_pAnnotHandlerMgr = NULL;
    delete m_pActionHandler;
    m_pActionHandler = NULL;
}

int CPDFDoc_Environment::JS_appAlert(const FX_WCHAR* Msg, const FX_WCHAR* Title, FX_UINT Type, FX_UINT Icon)
{
#ifdef _V8_SUPPORT_
    if (m_pInfo && m_pInfo->m_pJsPlatform && m_pInfo->m_pJsPlatform->app_alert)
    {
        CFX_ByteString bsMsg = CFX_WideString(Msg).UTF16LE_Encode();
        CFX_ByteString bsTitle = CFX_WideString(Title).UTF16LE_Encode();
        FPDF_WIDESTRING pMsg = (FPDF_WIDESTRING)bsMsg.GetBuffer(bsMsg.GetLength());
        FPDF_WIDESTRING pTitle = (FPDF_WIDESTRING)bsTitle.GetBuffer(bsTitle.GetLength());
        int ret = m_pInfo->m_pJsPlatform->app_alert(m_pInfo->m_pJsPlatform, pMsg, pTitle, Type, Icon);
        bsMsg.ReleaseBuffer();
        bsTitle.ReleaseBuffer();
        return ret;
    }
#endif  // ~ _V8_SUPPORT_
    return -1;
}

int CPDFDoc_Environment::JS_appResponse(const FX_WCHAR* Question, const FX_WCHAR* Title, const FX_WCHAR* Default,
                                        const FX_WCHAR* cLabel, FPDF_BOOL bPassword, void* response, int length)
{
#ifdef _V8_SUPPORT_
    if (m_pInfo && m_pInfo->m_pJsPlatform && m_pInfo->m_pJsPlatform->app_response)
    {
        CFX_ByteString bsQuestion = CFX_WideString(Question).UTF16LE_Encode();
        CFX_ByteString bsTitle = CFX_WideString(Title).UTF16LE_Encode();
        CFX_ByteString bsDefault = CFX_WideString(Default).UTF16LE_Encode();
        CFX_ByteString bsLabel = CFX_WideString(cLabel).UTF16LE_Encode();
        FPDF_WIDESTRING pQuestion = (FPDF_WIDESTRING)bsQuestion.GetBuffer(bsQuestion.GetLength());
        FPDF_WIDESTRING pTitle = (FPDF_WIDESTRING)bsTitle.GetBuffer(bsTitle.GetLength());
        FPDF_WIDESTRING pDefault = (FPDF_WIDESTRING)bsDefault.GetBuffer(bsDefault.GetLength());
        FPDF_WIDESTRING pLabel = (FPDF_WIDESTRING)bsLabel.GetBuffer(bsLabel.GetLength());
        int ret = m_pInfo->m_pJsPlatform->app_response(m_pInfo->m_pJsPlatform, pQuestion, pTitle,
                                                       pDefault, pLabel, bPassword, response, length);
        bsQuestion.ReleaseBuffer();
        bsTitle.ReleaseBuffer();
        bsDefault.ReleaseBuffer();
        bsLabel.ReleaseBuffer();
        return ret;
    }
#endif  // ~ _V8_SUPPORT_
    return -1;
}

CFX_WideString CPDFDoc_Environment::JS_fieldBrowse()
{
#ifdef _V8_SUPPORT_
    if (!m_pInfo ||
        !m_pInfo->m_pJsPlatform ||
        !m_pInfo->m_pJsPlatform->Field_browse) {
        return L"";
    }

    const int nRequiredLen = m_pInfo->m_pJsPlatform->Field_browse(
            m_pInfo->m_pJsPlatform, nullptr, 0);
    if (nRequiredLen <= 0)
        return L"";

    nonstd::unique_ptr<char[]> pBuff(new char[nRequiredLen]);
    memset(pBuff.get(), 0, nRequiredLen);
    const int nActualLen = m_pInfo->m_pJsPlatform->Field_browse(
            m_pInfo->m_pJsPlatform, pBuff.get(), nRequiredLen);
    if (nActualLen <= 0 || nActualLen > nRequiredLen)
        return L"";

    CFX_ByteString bsRet = CFX_ByteString(pBuff.get(), nActualLen);
    CFX_WideString wsRet = CFX_WideString::FromLocal(bsRet);
    return wsRet;
#else
    return L"";
#endif  // ~ _V8_SUPPORT_
}

CFX_WideString CPDFDoc_Environment::JS_docGetFilePath()
{
#ifdef _V8_SUPPORT_
    if (!m_pInfo ||
        !m_pInfo->m_pJsPlatform ||
        !m_pInfo->m_pJsPlatform->Doc_getFilePath) {
        return L"";
    }

    const int nRequiredLen = m_pInfo->m_pJsPlatform->Doc_getFilePath(
            m_pInfo->m_pJsPlatform, nullptr, 0);
    if (nRequiredLen <= 0)
        return L"";

    nonstd::unique_ptr<char[]> pBuff(new char[nRequiredLen]);
    memset(pBuff.get(), 0, nRequiredLen);
    const int nActualLen = m_pInfo->m_pJsPlatform->Doc_getFilePath(
            m_pInfo->m_pJsPlatform, pBuff.get(), nRequiredLen);
    if (nActualLen <= 0 || nActualLen > nRequiredLen)
        return L"";

    CFX_ByteString bsRet = CFX_ByteString(pBuff.get(), nActualLen);
    CFX_WideString wsRet = CFX_WideString::FromLocal(bsRet);
    return wsRet;
#else
    return L"";
#endif  // ~ _V8_SUPPORT_
}

void CPDFDoc_Environment::JS_docSubmitForm(void* formData, int length, const FX_WCHAR* URL)
{
#ifdef _V8_SUPPORT_
    if (m_pInfo && m_pInfo->m_pJsPlatform && m_pInfo->m_pJsPlatform->Doc_submitForm)
    {
        CFX_ByteString bsDestination = CFX_WideString(URL).UTF16LE_Encode();
        FPDF_WIDESTRING pDestination = (FPDF_WIDESTRING)bsDestination.GetBuffer(bsDestination.GetLength());
        m_pInfo->m_pJsPlatform->Doc_submitForm(m_pInfo->m_pJsPlatform, formData, length, pDestination);
        bsDestination.ReleaseBuffer();
    }
#endif  // ~ _V8_SUPPORT_
}

void CPDFDoc_Environment::JS_docmailForm(void* mailData, int length, FPDF_BOOL bUI,
                                         const FX_WCHAR* To, const FX_WCHAR* Subject,
                                         const FX_WCHAR* CC, const FX_WCHAR* BCC,
                                         const FX_WCHAR* Msg)
{
#ifdef _V8_SUPPORT_
    if (m_pInfo && m_pInfo->m_pJsPlatform && m_pInfo->m_pJsPlatform->Doc_mail)
    {
        CFX_ByteString bsTo = CFX_WideString(To).UTF16LE_Encode();
        CFX_ByteString bsCC = CFX_WideString(Subject).UTF16LE_Encode();
        CFX_ByteString bsBcc = CFX_WideString(BCC).UTF16LE_Encode();
        CFX_ByteString bsSubject = CFX_WideString(Subject).UTF16LE_Encode();
        CFX_ByteString bsMsg = CFX_WideString(Msg).UTF16LE_Encode();
        FPDF_WIDESTRING pTo = (FPDF_WIDESTRING)bsTo.GetBuffer(bsTo.GetLength());
        FPDF_WIDESTRING pCC = (FPDF_WIDESTRING)bsCC.GetBuffer(bsCC.GetLength());
        FPDF_WIDESTRING pBcc = (FPDF_WIDESTRING)bsBcc.GetBuffer(bsBcc.GetLength());
        FPDF_WIDESTRING pSubject = (FPDF_WIDESTRING)bsSubject.GetBuffer(bsSubject.GetLength());
        FPDF_WIDESTRING pMsg = (FPDF_WIDESTRING)bsMsg.GetBuffer(bsMsg.GetLength());
        m_pInfo->m_pJsPlatform->Doc_mail(m_pInfo->m_pJsPlatform, mailData, length, bUI, pTo, pSubject,
                                         pCC, pBcc, pMsg);
        bsTo.ReleaseBuffer();
        bsCC.ReleaseBuffer();
        bsBcc.ReleaseBuffer();
        bsSubject.ReleaseBuffer();
        bsMsg.ReleaseBuffer();
    }
#endif  // ~ _V8_SUPPORT_
}

IFXJS_Runtime* CPDFDoc_Environment::GetJSRuntime()
{
#ifdef _V8_SUPPORT_
    if (!IsJSInitiated())
        return NULL;
    if (!m_pJSRuntime)
        m_pJSRuntime = m_pJSRuntimeFactory->NewJSRuntime(this);
    return m_pJSRuntime;
#else
    return NULL;
#endif  // ~ _V8_SUPPORT_
}

CPDFSDK_AnnotHandlerMgr* CPDFDoc_Environment::GetAnnotHandlerMgr()
{
    if (!m_pAnnotHandlerMgr)
        m_pAnnotHandlerMgr = new CPDFSDK_AnnotHandlerMgr(this);
    return m_pAnnotHandlerMgr;
}

CPDFSDK_ActionHandler* CPDFDoc_Environment::GetActionHander()
{
    if (!m_pActionHandler)
        m_pActionHandler = new CPDFSDK_ActionHandler(this);
    return m_pActionHandler;
}

CFFL_IFormFiller* CPDFDoc_Environment::GetIFormFiller()
{
    if (!m_pIFormFiller)
        m_pIFormFiller = new CFFL_IFormFiller(this);
    return m_pIFormFiller;
}

CPDFSDK_Document::CPDFSDK_Document(CPDF_Document* pDoc,CPDFDoc_Environment* pEnv) :
    m_pDoc(pDoc),
    m_pInterForm(nullptr),
    m_pFocusAnnot(nullptr),
    m_pEnv(pEnv),
    m_pOccontent(nullptr),
    m_bChangeMask(FALSE)
{
}

CPDFSDK_Document::~CPDFSDK_Document()
{
    for (auto& it : m_pageMap)
        delete it.second;
    m_pageMap.clear();

    delete m_pInterForm;
    m_pInterForm = nullptr;

    delete m_pOccontent;
    m_pOccontent = nullptr;
}

CPDFSDK_PageView* CPDFSDK_Document::GetPageView(CPDF_Page* pPDFPage, FX_BOOL ReNew)
{
    auto it = m_pageMap.find(pPDFPage);
    if (it != m_pageMap.end())
        return it->second;

    if (!ReNew)
        return nullptr;

    CPDFSDK_PageView* pPageView = new CPDFSDK_PageView(this, pPDFPage);
    m_pageMap[pPDFPage] = pPageView;
    // Delay to load all the annotations, to avoid endless loop.
    pPageView->LoadFXAnnots();
    return pPageView;
}

CPDFSDK_PageView* CPDFSDK_Document::GetCurrentView()
{
    CPDF_Page* pPage = (CPDF_Page*)m_pEnv->FFI_GetCurrentPage(m_pDoc);
    return pPage ? GetPageView(pPage, TRUE) : nullptr;
}

CPDFSDK_PageView* CPDFSDK_Document::GetPageView(int nIndex)
{
    CPDF_Page* pTempPage = (CPDF_Page*)m_pEnv->FFI_GetPage(m_pDoc, nIndex);
    if (!pTempPage)
        return nullptr;

    auto it = m_pageMap.find(pTempPage);
    return it->second;
}

void CPDFSDK_Document:: ProcJavascriptFun()
{
    CPDF_Document* pPDFDoc = GetDocument();
    CPDF_DocJSActions docJS(pPDFDoc);
    int iCount = docJS.CountJSActions();
    if (iCount < 1) return;
    for (int i = 0; i < iCount; i ++)
    {
        CFX_ByteString csJSName;
        CPDF_Action jsAction = docJS.GetJSAction(i, csJSName);
        if(m_pEnv->GetActionHander())
            m_pEnv->GetActionHander()->DoAction_JavaScript(jsAction,CFX_WideString::FromLocal(csJSName),this);
    }

}

FX_BOOL CPDFSDK_Document::ProcOpenAction()
{
    if(!m_pDoc)
        return FALSE;

    CPDF_Dictionary* pRoot = m_pDoc->GetRoot();
    if (!pRoot)
        return FALSE;

    CPDF_Object* pOpenAction = pRoot->GetDict("OpenAction");
    if(!pOpenAction)
        pOpenAction = pRoot->GetArray("OpenAction");

    if(!pOpenAction)
        return FALSE;

    if(pOpenAction->GetType()==PDFOBJ_ARRAY)
        return TRUE;

    if(pOpenAction->GetType()==PDFOBJ_DICTIONARY)
    {
        CPDF_Dictionary * pDict=(CPDF_Dictionary*)pOpenAction;
        CPDF_Action action(pDict);
        if(m_pEnv->GetActionHander())
            m_pEnv->GetActionHander()->DoAction_DocOpen(action, this);
        return TRUE;
    }
    return FALSE;
}

CPDF_OCContext* CPDFSDK_Document::GetOCContext()
{
    if(!m_pOccontent)
        m_pOccontent = new CPDF_OCContext(m_pDoc);
    return m_pOccontent;
}

void CPDFSDK_Document::ReMovePageView(CPDF_Page* pPDFPage)
{
    auto it = m_pageMap.find(pPDFPage);
    if (it == m_pageMap.end())
        return;

    CPDFSDK_PageView* pPageView = it->second;
    if (pPageView->IsLocked())
        return;

    delete pPageView;
    m_pageMap.erase(it);
}

CPDF_Page * CPDFSDK_Document::GetPage(int nIndex)
{
    CPDF_Page * pTempPage = (CPDF_Page*)m_pEnv->FFI_GetPage(m_pDoc,nIndex);
    if(!pTempPage)
        return NULL;
    return pTempPage;
}

CPDFSDK_InterForm* CPDFSDK_Document::GetInterForm()
{
    if(!m_pInterForm)
        m_pInterForm = new CPDFSDK_InterForm(this);
    return m_pInterForm;
}

void CPDFSDK_Document::UpdateAllViews(CPDFSDK_PageView* pSender, CPDFSDK_Annot* pAnnot)
{
    for (const auto& it : m_pageMap) {
        CPDFSDK_PageView* pPageView = it.second;
        if (pPageView != pSender) {
            pPageView->UpdateView(pAnnot);
        }
    }
}

CPDFSDK_Annot* CPDFSDK_Document::GetFocusAnnot()
{
    return m_pFocusAnnot;
}

FX_BOOL CPDFSDK_Document::SetFocusAnnot(CPDFSDK_Annot* pAnnot,FX_UINT nFlag)
{

    if(m_pFocusAnnot==pAnnot) return TRUE;

    if(m_pFocusAnnot)
    {
        if(!KillFocusAnnot(nFlag) ) return FALSE;
    }
    CPDFSDK_PageView* pPageView = pAnnot->GetPageView();
    if(pAnnot && pPageView->IsValid())
    {
        CPDFSDK_AnnotHandlerMgr *pAnnotHandler=m_pEnv->GetAnnotHandlerMgr();

        if(pAnnotHandler&&!m_pFocusAnnot)
        {
            if (!pAnnotHandler->Annot_OnSetFocus(pAnnot,nFlag))
                return FALSE;
            if(!m_pFocusAnnot)
            {
                m_pFocusAnnot=pAnnot;
                return TRUE;
            }
        }
    }
    return FALSE;
}

FX_BOOL CPDFSDK_Document::KillFocusAnnot(FX_UINT nFlag)
{
    if(m_pFocusAnnot)
    {
        CPDFSDK_AnnotHandlerMgr *pAnnotHandler=m_pEnv->GetAnnotHandlerMgr();
        if(pAnnotHandler)
        {
            CPDFSDK_Annot* pFocusAnnot = m_pFocusAnnot;
            m_pFocusAnnot = NULL;
            if(pAnnotHandler->Annot_OnKillFocus(pFocusAnnot, nFlag))
            {

                if(pFocusAnnot->GetType() == FX_BSTRC("Widget"))
                {
                    CPDFSDK_Widget* pWidget = (CPDFSDK_Widget*)pFocusAnnot;
                    int nFieldType = pWidget->GetFieldType();
                    if(FIELDTYPE_TEXTFIELD == nFieldType || FIELDTYPE_COMBOBOX == nFieldType)
                        m_pEnv->FFI_OnSetFieldInputFocus(NULL, NULL, 0, FALSE);
                }

                if(!m_pFocusAnnot)
                    return TRUE;
            }
            else
            {
                m_pFocusAnnot = pFocusAnnot;
            }
        }
    }
    return FALSE;
}

void CPDFSDK_Document::OnCloseDocument()
{
    KillFocusAnnot();
}

FX_BOOL CPDFSDK_Document::GetPermissions(int nFlag)
{
    FX_DWORD dwPermissions = m_pDoc->GetUserPermissions();
    return dwPermissions&nFlag;
}

IFXJS_Runtime * CPDFSDK_Document::GetJsRuntime()
{
    ASSERT(m_pEnv!=NULL);
    return m_pEnv->GetJSRuntime();
}

CFX_WideString  CPDFSDK_Document::GetPath()
{
    ASSERT(m_pEnv != NULL);
    return m_pEnv->JS_docGetFilePath();
}


CPDFSDK_PageView::CPDFSDK_PageView(CPDFSDK_Document* pSDKDoc,CPDF_Page* page):m_page(page),m_pSDKDoc(pSDKDoc)
{
    CPDFSDK_InterForm* pInterForm = pSDKDoc->GetInterForm();
    if(pInterForm)
    {
        CPDF_InterForm* pPDFInterForm = pInterForm->GetInterForm();
        pPDFInterForm->FixPageFields(page);
    }
        m_page->SetPrivateData((void*)m_page, (void*)this, NULL);
    m_fxAnnotArray.RemoveAll();

    m_bEnterWidget = FALSE;
    m_bExitWidget = FALSE;
    m_bOnWidget = FALSE;
    m_CaptureWidget = NULL;
    m_bValid = FALSE;
        m_bLocked = FALSE;
        m_bTakeOverPage = FALSE;
}

CPDFSDK_PageView::~CPDFSDK_PageView()
{
    CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
    int nAnnotCount = m_fxAnnotArray.GetSize();

    for (int i=0; i<nAnnotCount; i++)
    {
        CPDFSDK_Annot* pAnnot = (CPDFSDK_Annot*)m_fxAnnotArray.GetAt(i);
        //if there is a focused annot on the page, we should kill the focus first.
        if(pAnnot == m_pSDKDoc->GetFocusAnnot())
            KillFocusAnnot();
        CPDFSDK_AnnotHandlerMgr* pAnnotHandlerMgr = pEnv->GetAnnotHandlerMgr();
        ASSERT(pAnnotHandlerMgr);
        pAnnotHandlerMgr->ReleaseAnnot(pAnnot);
    }
    m_fxAnnotArray.RemoveAll();

        delete m_pAnnotList;
        m_pAnnotList = NULL;

        m_page->RemovePrivateData((void*)m_page);
        if(m_bTakeOverPage) {
            delete m_page;
        }
}

void CPDFSDK_PageView::PageView_OnDraw(CFX_RenderDevice* pDevice, CPDF_Matrix* pUser2Device,CPDF_RenderOptions* pOptions)
{
    m_curMatrix = *pUser2Device;

    CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
    CPDFSDK_AnnotIterator annotIterator(this, TRUE);
    CPDFSDK_Annot* pSDKAnnot = nullptr;
    int index = -1;
    while ((pSDKAnnot = annotIterator.Next(index))) {
        CPDFSDK_AnnotHandlerMgr* pAnnotHandlerMgr = pEnv->GetAnnotHandlerMgr();
        ASSERT(pAnnotHandlerMgr);
        pAnnotHandlerMgr->Annot_OnDraw(
            this, pSDKAnnot, pDevice, pUser2Device, 0);
    }
}

CPDF_Annot* CPDFSDK_PageView::GetPDFAnnotAtPoint(FX_FLOAT pageX, FX_FLOAT pageY)
{

    int nCount = m_pAnnotList->Count();
    for(int i = 0 ; i<nCount; i++)
    {
        CPDF_Annot* pAnnot = m_pAnnotList->GetAt(i);
        CFX_FloatRect annotRect;
        pAnnot->GetRect(annotRect);
        if(annotRect.Contains(pageX, pageY))
            return pAnnot;
    }
    return NULL;
}

CPDF_Annot* CPDFSDK_PageView::GetPDFWidgetAtPoint(FX_FLOAT pageX, FX_FLOAT pageY)
{

    int nCount = m_pAnnotList->Count();
    for(int i = 0 ; i<nCount; i++)
    {
        CPDF_Annot* pAnnot = m_pAnnotList->GetAt(i);
        if(pAnnot->GetSubType() == "Widget")
        {
            CFX_FloatRect annotRect;
            pAnnot->GetRect(annotRect);
            if(annotRect.Contains(pageX, pageY))
                return pAnnot;
        }
    }
    return NULL;
}

CPDFSDK_Annot* CPDFSDK_PageView::GetFXAnnotAtPoint(FX_FLOAT pageX, FX_FLOAT pageY)
{

    CPDFSDK_AnnotIterator annotIterator(this, FALSE);
    CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
    CPDFSDK_AnnotHandlerMgr* pAnnotMgr = pEnv->GetAnnotHandlerMgr();
    CPDFSDK_Annot* pSDKAnnot = NULL;
    int index = -1;
    while((pSDKAnnot = annotIterator.Next(index)))
    {
        CPDF_Rect rc = pAnnotMgr->Annot_OnGetViewBBox(this, pSDKAnnot);
        if(rc.Contains(pageX, pageY))
            return pSDKAnnot;
    }

    return NULL;
}

CPDFSDK_Annot* CPDFSDK_PageView::GetFXWidgetAtPoint(FX_FLOAT pageX, FX_FLOAT pageY)
{

    CPDFSDK_AnnotIterator annotIterator(this, FALSE);
    CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
    CPDFSDK_AnnotHandlerMgr* pAnnotMgr = pEnv->GetAnnotHandlerMgr();
    CPDFSDK_Annot* pSDKAnnot = NULL;
    int index = -1;
    while((pSDKAnnot = annotIterator.Next(index)))
    {
        if(pSDKAnnot->GetType() == "Widget")
        {
            pAnnotMgr->Annot_OnGetViewBBox(this, pSDKAnnot);
            CPDF_Point point(pageX, pageY);
            if (pAnnotMgr->Annot_OnHitTest(this, pSDKAnnot, point))
                return pSDKAnnot;
        }
    }

    return NULL;
}


FX_BOOL CPDFSDK_PageView::Annot_HasAppearance(CPDF_Annot* pAnnot)
{
    CPDF_Dictionary* pAnnotDic = pAnnot->GetAnnotDict();
    if(pAnnotDic)
        return  pAnnotDic->KeyExist("AS");
    return FALSE;
}

CPDFSDK_Annot*  CPDFSDK_PageView::AddAnnot(CPDF_Annot * pPDFAnnot)
{
    CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
    ASSERT(pEnv);
    CPDFSDK_AnnotHandlerMgr * pAnnotHandler= pEnv->GetAnnotHandlerMgr();

    CPDFSDK_Annot* pSDKAnnot =NULL;

    if(pAnnotHandler)
    {
        pSDKAnnot = pAnnotHandler->NewAnnot(pPDFAnnot, this);
    }
    if(!pSDKAnnot)
        return NULL;

    m_fxAnnotArray.Add(pSDKAnnot);

    if(pAnnotHandler)
    {
        pAnnotHandler->Annot_OnCreate(pSDKAnnot);

    }

     return pSDKAnnot;
}

CPDFSDK_Annot* CPDFSDK_PageView::AddAnnot(CPDF_Dictionary * pDict)
{
    return pDict ? AddAnnot(pDict->GetString("Subtype"), pDict) : nullptr;
}

CPDFSDK_Annot* CPDFSDK_PageView::AddAnnot(const FX_CHAR* lpSubType,CPDF_Dictionary * pDict)
{
    return NULL;
}

FX_BOOL  CPDFSDK_PageView::DeleteAnnot(CPDFSDK_Annot* pAnnot)
{
    return FALSE;
}

CPDF_Document* CPDFSDK_PageView::GetPDFDocument()
{
    if(m_page)
    {
        return m_page->m_pDocument;
    }
    return NULL;
}

int CPDFSDK_PageView::CountAnnots()
{
    return m_pAnnotList->Count();
}

CPDFSDK_Annot*  CPDFSDK_PageView::GetAnnot(int nIndex)
{
    int nCount = m_fxAnnotArray.GetSize();
    if ( nIndex < 0 || nIndex >= nCount )
    {
        return NULL;
    }

    return (CPDFSDK_Annot*)m_fxAnnotArray.GetAt(nIndex);
}

CPDFSDK_Annot* CPDFSDK_PageView::GetAnnotByDict(CPDF_Dictionary * pDict)
{
    int nCount = m_fxAnnotArray.GetSize();
    for(int i=0; i<nCount; i++)
    {
        CPDFSDK_Annot* pAnnot = (CPDFSDK_Annot*)m_fxAnnotArray.GetAt(i);
        if (pDict == pAnnot->GetPDFAnnot()->GetAnnotDict())
            return pAnnot;
    }
    return NULL;
}

FX_BOOL CPDFSDK_PageView::OnLButtonDown(const CPDF_Point & point, FX_UINT nFlag)
{
    CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
    ASSERT(pEnv);
    CPDFSDK_Annot* pFXAnnot = GetFXWidgetAtPoint(point.x, point.y);
    if(!pFXAnnot)
    {
        KillFocusAnnot(nFlag);
    }
    else
    {
        CPDFSDK_AnnotHandlerMgr* pAnnotHandlerMgr = pEnv->GetAnnotHandlerMgr();
        ASSERT(pAnnotHandlerMgr);

        FX_BOOL bRet = pAnnotHandlerMgr->Annot_OnLButtonDown(this, pFXAnnot, nFlag,point);
        if(bRet)
        {
            SetFocusAnnot(pFXAnnot);
        }
        return bRet;
    }
    return FALSE;
}


FX_BOOL CPDFSDK_PageView::OnLButtonUp(const CPDF_Point & point, FX_UINT nFlag)
{
    CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
    ASSERT(pEnv);
    CPDFSDK_AnnotHandlerMgr* pAnnotHandlerMgr = pEnv->GetAnnotHandlerMgr();
    ASSERT(pAnnotHandlerMgr);
    CPDFSDK_Annot* pFXAnnot = GetFXWidgetAtPoint(point.x, point.y);
    CPDFSDK_Annot* pFocusAnnot = GetFocusAnnot();
    FX_BOOL bRet  = FALSE;
    if(pFocusAnnot && pFocusAnnot != pFXAnnot)
    {
        //Last focus Annot gets a chance to handle the event.
        bRet = pAnnotHandlerMgr->Annot_OnLButtonUp(this, pFocusAnnot, nFlag,point);
    }
    if(pFXAnnot && !bRet)
    {
        bRet = pAnnotHandlerMgr->Annot_OnLButtonUp(this, pFXAnnot, nFlag,point);
        return bRet;
    }
    return bRet;
}

FX_BOOL CPDFSDK_PageView::OnMouseMove(const CPDF_Point & point, int nFlag)
{

    CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
    CPDFSDK_AnnotHandlerMgr* pAnnotHandlerMgr = pEnv->GetAnnotHandlerMgr();
    ASSERT(pAnnotHandlerMgr);
    if(CPDFSDK_Annot* pFXAnnot = GetFXWidgetAtPoint(point.x, point.y))
    {
        if(m_CaptureWidget && m_CaptureWidget != pFXAnnot)
        {
            m_bExitWidget = TRUE;
            m_bEnterWidget = FALSE;
            pAnnotHandlerMgr->Annot_OnMouseExit(this, m_CaptureWidget, nFlag);
        }
        m_CaptureWidget = (CPDFSDK_Widget*)pFXAnnot;
        m_bOnWidget = TRUE;
        if(!m_bEnterWidget)
        {
            m_bEnterWidget = TRUE;
            m_bExitWidget = FALSE;
            pAnnotHandlerMgr->Annot_OnMouseEnter(this, pFXAnnot,nFlag);
        }
        pAnnotHandlerMgr->Annot_OnMouseMove(this, pFXAnnot, nFlag, point);
        return TRUE;
    }
    if(m_bOnWidget)
    {
        m_bOnWidget = FALSE;
        m_bExitWidget = TRUE;
        m_bEnterWidget = FALSE;
        if(m_CaptureWidget)
        {
            pAnnotHandlerMgr->Annot_OnMouseExit(this, m_CaptureWidget, nFlag);
            m_CaptureWidget = NULL;
        }
    }
    return FALSE;
}

FX_BOOL CPDFSDK_PageView::OnMouseWheel(double deltaX, double deltaY,const CPDF_Point& point, int nFlag)
{
    if(CPDFSDK_Annot* pAnnot = GetFXWidgetAtPoint(point.x, point.y))
    {
        CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
        CPDFSDK_AnnotHandlerMgr* pAnnotHandlerMgr = pEnv->GetAnnotHandlerMgr();
        ASSERT(pAnnotHandlerMgr);
        return pAnnotHandlerMgr->Annot_OnMouseWheel(this, pAnnot, nFlag, (int)deltaY, point);
    }
    return FALSE;

}

FX_BOOL CPDFSDK_PageView::OnChar(int nChar, FX_UINT nFlag)
{
    if(CPDFSDK_Annot* pAnnot = GetFocusAnnot())
    {
        CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
        CPDFSDK_AnnotHandlerMgr* pAnnotHandlerMgr = pEnv->GetAnnotHandlerMgr();
        ASSERT(pAnnotHandlerMgr);
        return pAnnotHandlerMgr->Annot_OnChar(pAnnot, nChar, nFlag);
    }

    return FALSE;
}

FX_BOOL CPDFSDK_PageView::OnKeyDown(int nKeyCode, int nFlag)
{
    if(CPDFSDK_Annot* pAnnot = GetFocusAnnot())
    {
        CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
        CPDFSDK_AnnotHandlerMgr* pAnnotHandlerMgr = pEnv->GetAnnotHandlerMgr();
        ASSERT(pAnnotHandlerMgr);
        return pAnnotHandlerMgr->Annot_OnKeyDown(pAnnot, nKeyCode, nFlag);
    }
    return FALSE;
}

FX_BOOL CPDFSDK_PageView::OnKeyUp(int nKeyCode, int nFlag)
{
//  if(CPDFSDK_Annot* pAnnot = GetFocusAnnot())
//  {
//      CFFL_IFormFiller* pIFormFiller = g_pFormFillApp->GetIFormFiller();
//      return pIFormFiller->OnKeyUp(pAnnot, nKeyCode, nFlag);
//  }
    return FALSE;
}

extern void CheckUnSupportAnnot(CPDF_Document * pDoc, CPDF_Annot* pPDFAnnot);

void CPDFSDK_PageView::LoadFXAnnots()
{
    CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();

    FX_BOOL enableAPUpdate = CPDF_InterForm::UpdatingAPEnabled();
    //Disable the default AP construction.
    CPDF_InterForm::EnableUpdateAP(FALSE);
    m_pAnnotList = new CPDF_AnnotList(m_page);
    CPDF_InterForm::EnableUpdateAP(enableAPUpdate);
    int nCount = m_pAnnotList->Count();
        SetLock(TRUE);
    for(int i=0; i<nCount; i++)
    {
        CPDF_Annot* pPDFAnnot = m_pAnnotList->GetAt(i);
        CPDF_Document * pDoc = GetPDFDocument();

        CheckUnSupportAnnot(pDoc, pPDFAnnot);

        CPDFSDK_AnnotHandlerMgr* pAnnotHandlerMgr = pEnv->GetAnnotHandlerMgr();
        ASSERT(pAnnotHandlerMgr != NULL);

        if(pAnnotHandlerMgr)
        {
            CPDFSDK_Annot* pAnnot = pAnnotHandlerMgr->NewAnnot(pPDFAnnot, this);
            if(!pAnnot)
                continue;
            m_fxAnnotArray.Add(pAnnot);

            pAnnotHandlerMgr->Annot_OnLoad(pAnnot);
        }

    }
        SetLock(FALSE);
}

void    CPDFSDK_PageView::UpdateRects(CFX_RectArray& rects)
{
    for(int i=0; i<rects.GetSize(); i++)
    {
        CPDF_Rect rc = rects.GetAt(i);
        CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
        pEnv->FFI_Invalidate(m_page, rc.left, rc.top, rc.right, rc.bottom);
    }
}

void CPDFSDK_PageView::UpdateView(CPDFSDK_Annot* pAnnot)
{
    CPDF_Rect rcWindow = pAnnot->GetRect();
    CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
    pEnv->FFI_Invalidate(
        m_page, rcWindow.left, rcWindow.top, rcWindow.right, rcWindow.bottom);
}

int CPDFSDK_PageView::GetPageIndex()
{
    if(m_page)
    {
        CPDF_Dictionary* pDic = m_page->m_pFormDict;
        CPDF_Document* pDoc = m_pSDKDoc->GetDocument();
        if(pDoc && pDic)
        {
            return pDoc->GetPageIndex(pDic->GetObjNum());
        }
    }
    return -1;
}

FX_BOOL CPDFSDK_PageView::IsValidAnnot(void* p)
{
    if (p == NULL) return FALSE;
    int iCount = m_pAnnotList->Count();
    for (int i = 0; i < iCount; i++)
    {
        if (m_pAnnotList->GetAt(i) == p)
            return TRUE;
    }
    return FALSE;
}


CPDFSDK_Annot* CPDFSDK_PageView::GetFocusAnnot()
{
    CPDFSDK_Annot* pFocusAnnot = m_pSDKDoc->GetFocusAnnot();
    if(!pFocusAnnot)
        return NULL;

    for(int i=0; i<m_fxAnnotArray.GetSize(); i++)
    {
        CPDFSDK_Annot* pAnnot = (CPDFSDK_Annot*)m_fxAnnotArray.GetAt(i);
        if(pAnnot == pFocusAnnot)
            return pAnnot;
    }
    return NULL;
}
